Skip to content

Fix BERT model loading by removing deprecated cache_dir parameter#12

Closed
Hrushik777 wants to merge 2 commits intoOpenNyAI:masterfrom
Hrushik777:master
Closed

Fix BERT model loading by removing deprecated cache_dir parameter#12
Hrushik777 wants to merge 2 commits intoOpenNyAI:masterfrom
Hrushik777:master

Conversation

@Hrushik777
Copy link
Copy Markdown

Removed deprecated cache_dir parameter from BertModel.from_pretrained().
This parameter is no longer required in newer HuggingFace versions and causes loading issues.

@prathameshk
Copy link
Copy Markdown
Contributor

Closing this PR as the fix has been incorporated into v0.0.15 (published on PyPI). The deprecated cache_dir parameter has been removed from BertModel.from_pretrained() in the new release. Thank you for identifying this issue!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants